99f9c624ffa4abbeb6c3fb0083065465794fe790,src/main/java/org/apache/pirk/test/utils/BaseTests.java,BaseTests,testSRCIPQueryNoFilter,#ArrayList#FileSystem#boolean#boolean#number#,403
Before Change
qrJSON.setMapping(Inputs.DSTIP, dataMap.get(Inputs.DSTIP));
qrJSON.setMapping(Inputs.SRCIP, dataMap.get(Inputs.SRCIP));
qrJSON.setMapping(Inputs.IPS, parseArray(dataMap, Inputs.IPS, true));
qrJSON.setMapping(QueryResponseJSON.SELECTOR, QueryUtils.getSelectorByQueryTypeJSON(Inputs.DNS_SRCIP_QUERY_NO_FILTER, dataMap));
correctResults.add(qrJSON);
}
++i;
After Change
{
logger.info("Running testSRCIPQueryNoFilter(): ");
QuerySchema qSchema = LoadQuerySchemas.getSchema(Inputs.DNS_SRCIP_QUERY_NO_FILTER);
ArrayList<QueryResponseJSON> results;
int numExpectedResults = 3;
if (isDistributed)
{
results = DistTestSuite.performQuery(Inputs.DNS_SRCIP_QUERY_NO_FILTER, selectorsIP, fs, isSpark, numThreads);
}
else
{
results = StandaloneQuery.performStandaloneQuery(dataElements, Inputs.DNS_SRCIP_QUERY_NO_FILTER, selectorsIP, numThreads, false);
}
printResultList(results);
if (results.size() != numExpectedResults)
{
fail("results.size() = " + results.size() + " -- must equal " + numExpectedResults);
}
ArrayList<QueryResponseJSON> correctResults = new ArrayList<>();
int i = 0;
while (i < dataElements.size())
{
JSONObject dataMap = dataElements.get(i);
boolean addElement = false;
if (dataMap.get(Inputs.SRCIP).toString().equals("55.55.55.55") || dataMap.get(Inputs.SRCIP).toString().equals("5.6.7.8"))
{
addElement = true;
}
if (addElement)
{
// Form the correct result QueryResponseJSON object
QueryResponseJSON qrJSON = new QueryResponseJSON();
qrJSON.setMapping(QueryResponseJSON.QUERY_ID, queryNum);
qrJSON.setMapping(QueryResponseJSON.QUERY_NAME, Inputs.DNS_SRCIP_QUERY_NO_FILTER + "_" + queryNum);
qrJSON.setMapping(QueryResponseJSON.EVENT_TYPE, Inputs.DNS_SRCIP_QUERY_NO_FILTER);
qrJSON.setMapping(Inputs.QNAME, parseString(dataMap, Inputs.QNAME));
qrJSON.setMapping(Inputs.DSTIP, dataMap.get(Inputs.DSTIP));
qrJSON.setMapping(Inputs.SRCIP, dataMap.get(Inputs.SRCIP));
qrJSON.setMapping(Inputs.IPS, parseArray(dataMap, Inputs.IPS, true));
qrJSON.setMapping(QueryResponseJSON.SELECTOR, QueryUtils.getSelectorByQueryTypeJSON(qSchema, dataMap));
correctResults.add(qrJSON);
}
++i;